#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <string.h>
#include <stdlib.h>
#include <fcntl.h>

#define MAXLINE 1024

int main() {
    // 创建socket对象
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0) {
        perror("socket creation failed");
        return 1;
    }

    // 创建网络通信对象
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(8888);
    addr.sin_addr.s_addr = inet_addr("127.0.0.1");

    // 连接服务器
    if (connect(sockfd, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
        perror("ERROR connecting");
        exit(1);
    }

    // 请求用户输入文件路径
    char filepath[MAXLINE];
    printf("请输入要请求的文件路径: ");
    if (fgets(filepath, MAXLINE, stdin) == NULL) {
        perror("无法读取文件路径");
        close(sockfd);
        return 1;
    }

    // 删除换行符（如果有的话）
    filepath[strcspn(filepath, "\n")] = '\0';

    // 发送文件名请求
    if (send(sockfd, filepath, strlen(filepath), 0) < 0) {
        perror("发送失败");
        close(sockfd);
        return 1;
    }
    printf("已发送文件名请求到服务器\n");

    // 接收服务器端的响应
    char recv_buffer[MAXLINE];
    ssize_t n;
    FILE *file = fopen("received_file.txt", "wb");
    if (file == NULL) {
        perror("无法创建本地文件");
        close(sockfd);
        return 1;
    }

    while ((n = recv(sockfd, recv_buffer, MAXLINE, 0)) > 0) {
        fwrite(recv_buffer, 1, n, file);
    }

    fclose(file);
    close(sockfd);

    if (n < 0) {
        perror("接收失败");
        return 1;
    } else if (n == 0) {
        printf("文件传输完成\n");
    }

    return 0;
}
